﻿//3217. 从链表中移除在数组中存在的节点
//给你一个整数数组 nums 和一个链表的头节点 head。从链表中移除所有存在于 nums 中的节点后，返回修改后的链表的头节点。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* modifiedList(vector<int>& nums, ListNode* head)
    {
        unordered_set<int> cnt;
        for (auto x : nums)
            cnt.insert(x);
        ListNode hair(0);
        hair.next = head;
        auto p = &hair;
        while (p->next)
        {
            auto n = p->next;
            if (cnt.count(n->val))
            {
                p->next = n->next;
                //delete n;
            }
            else
            {
                p = n;
            }
        }
        return hair.next;
    }
};